from ship_envs.environment.single_ship_continuous_env import SingleShipContinuousEnv


def render_test():
    env = SingleShipContinuousEnv()
    for _ in range(3):
        env.reset()
        steps = 0
        while True:
            steps += 1
            action = env.action_space.sample()
            obs, reward, done, info = env.step(action)
            env.render()
            if done:
                print(steps)
                break


if __name__ == '__main__':
    render_test()